//
//  IsSubStructure.swift
//  LeetCodeSummary
//
//  Created by 彭西西 on 2020/7/28.
//  Copyright © 2020 WangYonghe. All rights reserved.
//  剑指 Offer 26. 树的子结构

import UIKit

/*
 剑指 Offer 26. 树的子结构
 
 输入两棵二叉树A和B，判断B是不是A的子结构。(约定空树不是任意一个树的子结构)

 B是A的子结构， 即 A中有出现和B相同的结构和节点值。

 例如:
 给定的树 A:

      3
     / \
    4   5
   / \
  1   2
 给定的树 B：

    4
   /
  1
 返回 true，因为 B 与 A 的一个子树拥有相同的结构和节点值。

 示例 1：

 输入：A = [1,2,3], B = [3,1]
 输出：false
 示例 2：

 输入：A = [3,4,5,1,2], B = [4,1]
 输出：true
 限制：

 0 <= 节点个数 <= 10000
 */

class IsSubStructure: NSObject {
    func isSubStructure(_ A: TreeNode?, _ B: TreeNode?) -> Bool {

        guard A != nil else{
            return false
        }
        
        guard B != nil else{
            return false
        }
        
        if A?.val == B?.val && self.helper(A?.left, B?.left) == true && self.helper(A?.right, B?.right){
            return true
        }
        
        return self.isSubStructure(A?.left, B) || self.isSubStructure(A?.right, B)
    }
    
    func helper(_ A: TreeNode?, _ B: TreeNode?)->Bool{
        
        if B == nil {
            return true
        }
        if A == nil {
            return false
        }
        if A?.val == B?.val {
            return self.helper(A?.left, B?.left) && self.helper(A?.right, B?.right)
        }
        return false
    }
    
    
    /*
     recur函数>>>>
     
     终止条件：
     当节点B为空时，说明树B已经完成匹配(超过叶子节点)，因此返回true
     当节点A为空时，说明已经越过A叶子节点，匹配失败，返回false
     当节点A和B不同时，匹配失败，返回false
     
     返回值：
     判断A和B的左子节点是否相同 self.recur(A.left,B.left)
     判断A和B的右子节点是否相同 self.recur(A.right,B.right)
     */
    
    
    //更巧妙的写法
    func isSubStructure2(_ A: TreeNode?, _ B: TreeNode?) -> Bool {
        return (A != nil && B != nil) && (recur(A, B) || self.isSubStructure2(A?.left, B) || self.isSubStructure2(A?.right, B))
    }
    
    func recur(_ A: TreeNode?, _ B: TreeNode?)->Bool{
        
        if B == nil{
            return true
        }
        if A == nil || A?.val != B?.val{
            return false
        }
        return self.recur(A?.left, B?.left) && self.recur(A?.right, B?.right)
    }
}
